package netty.client;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;
import io.netty.handler.logging.LoggingHandler;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.SslContextBuilder;
import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
import io.netty.util.concurrent.Future;
import io.netty.util.concurrent.GenericFutureListener;
import lombok.extern.slf4j.Slf4j;

import java.util.concurrent.TimeUnit;

@Slf4j
public class SimpleMultiClient {
    static final boolean SSL = System.getProperty("ssl") != null;
    static final String HOST = System.getProperty("host", "127.0.0.1");

    static final int PORT = Integer.parseInt(System.getProperty("port", "8009"));
    static final int SIZE = Integer.parseInt(System.getProperty("size", "256"));

    private final Bootstrap b;

    public SimpleMultiClient() throws Exception {
        // Configure SSL.
        final SslContext sslCtx;
        if (SSL) {
            sslCtx = SslContextBuilder.forClient()
                    .trustManager(InsecureTrustManagerFactory.INSTANCE).build();
        } else {
            sslCtx = null;
        }

        EventLoopGroup group = new NioEventLoopGroup();
        try {
            b = new Bootstrap();
            b.group(group)
                    .channel(NioSocketChannel.class)
                    .handler(new LoggingHandler())
                    .handler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            ChannelPipeline p = ch.pipeline();
                            if (sslCtx != null) {
                                p.addLast(sslCtx.newHandler(ch.alloc(), HOST, PORT));
                            }
                            p.addLast(new StringDecoder());
                            p.addLast(new StringEncoder());
                            p.addLast(new SimpleMultiHandler());
                        }
                    });
        } finally {
            // group.shutdownGracefully();
        }
    }

    public static void main(String[] args) throws Exception {
        SimpleMultiClient client = new SimpleMultiClient();

        for (int i = 0; i < 100; i++) {
            client.connect();
        }
    }

    public ChannelFuture connect() throws Exception {
        // Make the connection attempt.
        ChannelFuture f = b.connect(HOST, PORT).sync();

        log.info("connect to {}:{}", HOST, PORT);

        f.addListener(new GenericFutureListener<Future<? super Void>>() {
            @Override
            public void operationComplete(Future<? super Void> future) throws Exception {
                if (future.isSuccess()) {
                    log.info("connect success");
                    // f.channel().writeAndFlush("abc");
                    f.channel().eventLoop().scheduleAtFixedRate(() -> {
                        f.channel().writeAndFlush("abc");
                    }, 3, 3, TimeUnit.SECONDS);
                } else {
                    log.info("connect fail");
                }
            }
        });

        // Wait until the connection is closed.
        // f.channel().closeFuture().sync();

        f.channel().closeFuture();

        log.info("connect to {}:{}", HOST, PORT);
        return f;
    }
}
